
"""
数据分析的状态
"""

from langgraph.graph import MessagesState
from typing import List, Dict
from models.base import BaseData
from models.userinfo import UserInfoData

class AnalyzeState(MessagesState):
    sessionId: str                # 用户sessionId
    query: str                    # 用户的问题描述
    userinfo: UserInfoData | None # 用户信息
    schemas: List[Dict]           # 全部数据表结构及其预览
    rewrites: List[str]           # 改写的问题列表
    datasets: List[BaseData]      # 问数查询出的数据表
    filtered: List[BaseData]      # 过滤出的数据表
